from method import *

if __name__ == '__main__':
    source_path = 'C:/myself/hw_ai/hw-smoke-detection/datasets/helmetData/train/xml/'
    label_path = 'C:/myself/hw_ai/hw-smoke-detection/datasets/helmetData/train/labels/'
    xml_to_yolotxt(source_path, label_path)
    
    source_path = 'C:/myself/hw_ai/hw-smoke-detection/datasets/helmetData/val/xml/'
    label_path = 'C:/myself/hw_ai/hw-smoke-detection/datasets/helmetData/val/labels/'
    xml_to_yolotxt(source_path, label_path)